package com.aube.plugin.gallery.preview;

import android.content.Context;
import android.support.v7.widget.LinearLayoutManager;
import android.support.v7.widget.RecyclerView;
import android.view.View;
import android.view.ViewGroup;

import com.huyn.baseframework.utils.Utils;

/**
 * Created by huyaonan on 16/9/18.
 */
public class SelfWrapLinearLayoutManager extends LinearLayoutManager {

    private int bottomOffset = 0;
    private OnHeightSizeListener mHeightListener;

    private int[] mMeasuredDimension = new int[2];

    public SelfWrapLinearLayoutManager(Context context) {
        super(context);
        bottomOffset = Utils.dip2px(context, 50);
    }

    public SelfWrapLinearLayoutManager(Context context, int orientation, boolean reverseLayout) {
        super(context, orientation, reverseLayout);
        bottomOffset = Utils.dip2px(context, 50);
    }

    public void setOnHeightSizeListener(OnHeightSizeListener listener) {
        this.mHeightListener = listener;
    }

    public void onMeasure(RecyclerView.Recycler recycler, RecyclerView.State state, int widthSpec, int heightSpec) {
//        final int widthMode = View.MeasureSpec.getMode(widthSpec);
//        final int heightMode = View.MeasureSpec.getMode(heightSpec);
        final int widthSize = View.MeasureSpec.getSize(widthSpec);
        final int heightSize = View.MeasureSpec.getSize(heightSpec);

        int width = 0;
        int height = 0;
        int count = getItemCount();
        for (int i = 0; i < count; i++) {
            measureScrapChild(recycler, i,
                    View.MeasureSpec.makeMeasureSpec(i, View.MeasureSpec.UNSPECIFIED),
                    View.MeasureSpec.makeMeasureSpec(i, View.MeasureSpec.UNSPECIFIED),
                    mMeasuredDimension);

            if (getOrientation() == HORIZONTAL) {
                width = width + mMeasuredDimension[0];
                height = mMeasuredDimension[1];
            } else {
                height = height + mMeasuredDimension[1];
                width = mMeasuredDimension[0];

                if(height > heightSize - bottomOffset)
                    break;
            }
        }

//        switch (widthMode) {
//            case View.MeasureSpec.EXACTLY:
//                width = widthSize;
//            case View.MeasureSpec.AT_MOST:
//            case View.MeasureSpec.UNSPECIFIED:
//        }
//
//        switch (heightMode) {
//            case View.MeasureSpec.EXACTLY:
//                height = heightSize;
//            case View.MeasureSpec.AT_MOST:
//            case View.MeasureSpec.UNSPECIFIED:
//        }

        int finalSize = height > heightSize - bottomOffset ? heightSize - bottomOffset: height;
//        Utils.sysout("----final size..." + height + "/" + heightSize);
        if(mHeightListener != null)
            mHeightListener.onHeightMeasured(finalSize);

        setMeasuredDimension(widthSize, finalSize);
    }

    private void measureScrapChild(RecyclerView.Recycler recycler, int position, int widthSpec, int heightSpec, int[] measuredDimension) {
        if (position < getItemCount()) {
            try {
                View view = recycler.getViewForPosition(position);//fix 动态添加时报IndexOutOfBoundsException
                if (view != null) {
                    RecyclerView.LayoutParams p = (RecyclerView.LayoutParams) view.getLayoutParams();
                    int childWidthSpec = ViewGroup.getChildMeasureSpec(widthSpec,
                            getPaddingLeft() + getPaddingRight(), p.width);
                    int childHeightSpec = ViewGroup.getChildMeasureSpec(heightSpec,
                            getPaddingTop() + getPaddingBottom(), p.height);
                    view.measure(childWidthSpec, childHeightSpec);
                    measuredDimension[0] = view.getMeasuredWidth() + p.leftMargin + p.rightMargin;
                    measuredDimension[1] = view.getMeasuredHeight() + p.bottomMargin + p.topMargin;
//                    recycler.recycleView(view);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    public interface OnHeightSizeListener {
        public void onHeightMeasured(int height);
    }

}
